package concursos.controladores;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import concursos.modelos.Arquivo;

public class ArquivoDataModel extends ListDataModel<Arquivo> implements SelectableDataModel<Arquivo>{
	
	public ArquivoDataModel(){}
	
	public ArquivoDataModel(List<Arquivo> data){
		super(data);
	}
	
	@Override
	public Arquivo getRowData(String rowKey) {
		@SuppressWarnings("unchecked")
		List<Arquivo> arquivos = (List<Arquivo>) getWrappedData();  
        
        for(Arquivo arquivo : arquivos) {  
            if(arquivo.getId() == Integer.parseInt(rowKey))  
                return arquivo;  
        }  
          
        return null;  
	}

	@Override
	public Object getRowKey(Arquivo arquivo) {
		return arquivo.getId();
	}

}
